"""
作者：Leagolas
日期：2023年05月30日
"""

import pymysql
import InitSystem
import Functions
import time


#功能选择界面
def selectFunction():
    while True:
        print("\n" * 1)
        print("---请选择需要的功能---")
        print("      1:点单")
        print("      2:入库")
        print("      3:清单管理")
        print("      4:销售分析")
        print("---请选择需要的功能---")
        command = input("请选择需求:> ")
        if command == '1':
            time.sleep(0.5)
            print('\n' * 1)
            Functions.function1()

        if command == '2':
            time.sleep(0.5)
            print('\n' * 1)
            Functions.function2()

        if command == '3':
            time.sleep(0.5)
            print('\n' * 1)
            Functions.function3()

        if command == '4':
            time.sleep(0.5)
            print('\n' * 1)
            Functions.function4()

        if command == '#':
            break

#欢迎界面
def welcome():
        while True:
            print("---欢迎使用库存管理系统---")
            print("        1:开始")
            print("        0:退出")
            print("---欢迎使用库存管理系统---")
            command = input("请选择需求:> ")
            if (command == '0' or command == '#'):
                break
            elif command == '1':
                selectFunction();

#检查数据库是否已存在，如不存在则进行初始化
def start():
    if not InitSystem.isInit():
        print("---检测到数据库尚未创建---")
        print("---开始初始化数据库---")
        InitSystem.initDatabase()
        print("---初始化成功---")
        time.sleep(2)
        print('\n' * 3)
        welcome()
    else:
        welcome()

#系统开始
start()